# Think内容过滤测试脚本

Write-Host "=== Think内容过滤功能测试 ===" -ForegroundColor Green

$baseUrl = "http://localhost:8082/api/ollama"

Write-Host "`n测试场景：发送可能包含think标签的问题" -ForegroundColor Yellow

# 测试问题，这些问题可能会触发AI的思考过程
$testPrompts = @(
    "你是谁？请详细介绍一下自己",
    "请解释一下什么是人工智能",
    "帮我分析一下Spring Boot的优势",
    "请思考并回答：为什么要使用微服务架构？"
)

foreach ($i in 0..($testPrompts.Length - 1)) {
    $prompt = $testPrompts[$i]
    Write-Host "`n--- 测试 $($i + 1): $prompt ---" -ForegroundColor Cyan
    
    try {
        $body = @{
            prompt = $prompt
        } | ConvertTo-Json

        Write-Host "发送请求..." -ForegroundColor Gray
        $response = Invoke-RestMethod -Uri "$baseUrl/chat" -Method POST -Body $body -ContentType "application/json" -TimeoutSec 30
        
        if ($response.success) {
            Write-Host "✅ 请求成功" -ForegroundColor Green
            Write-Host "响应长度: $($response.response.Length) 字符" -ForegroundColor Cyan
            
            # 检查响应中是否还包含think标签
            if ($response.response -match "<think>.*?</think>") {
                Write-Host "❌ 警告：响应中仍包含think标签内容！" -ForegroundColor Red
            } else {
                Write-Host "✅ 确认：响应中已成功过滤think标签内容" -ForegroundColor Green
            }
            
            # 显示响应的前100个字符
            $preview = if ($response.response.Length -gt 100) { 
                $response.response.Substring(0, 100) + "..." 
            } else { 
                $response.response 
            }
            Write-Host "响应预览: $preview" -ForegroundColor White
        } else {
            Write-Host "❌ 请求失败: $($response.error)" -ForegroundColor Red
        }
    } catch {
        Write-Host "❌ 请求异常: $($_.Exception.Message)" -ForegroundColor Red
    }
    
    Write-Host "等待2秒..." -ForegroundColor Gray
    Start-Sleep -Seconds 2
}

Write-Host "`n=== 过滤功能测试完成 ===" -ForegroundColor Green
Write-Host "`n💡 说明:" -ForegroundColor Magenta
Write-Host "- 如果看到 '✅ 确认：响应中已成功过滤think标签内容'，说明过滤功能正常工作" -ForegroundColor White
Write-Host "- 如果看到 '❌ 警告：响应中仍包含think标签内容'，说明需要检查过滤逻辑" -ForegroundColor White
Write-Host "- 你现在只会收到AI的最终回答，不会看到思考过程" -ForegroundColor White
